from django.db import models
from django.contrib.auth.models import AbstractUser
from django_mysql.models import EnumField


# Create your models here.

class User(AbstractUser):
    mobile = models.CharField(max_length=11, unique=True, verbose_name='手机号')
    email_active = models.BooleanField(default=False, verbose_name='邮箱验证状态')
    default_address = models.ForeignKey('Address', related_name='default_user', null=True, blank=True,
                                        on_delete=models.SET_NULL, verbose_name='默认地址')

    class Meta:
        db_table = 'user'
        verbose_name = '用户'
        verbose_name_plural = verbose_name

    def __str__(self):
        return self.username


class Broker(models.Model):
    NAME_LIST = (
        ('1', 'weibo'),
        ('2', 'gitee'),
    )
    name = EnumField(choices=NAME_LIST)
    user = models.ForeignKey(User, on_delete=models.CASCADE, related_name='broker', null=True)
    # name = models.TextChoices(('weibo', 'gitee'))
    mark = models.CharField(max_length=64, verbose_name='三方的唯一标示')

    class Meta:
        db_table = 'broker'
        verbose_name = '三方登录'
        verbose_name_plural = verbose_name

    def __str__(self):
        return f"{self.name} {self.user.name}"


class Address(models.Model):

    user = models.ForeignKey(User, on_delete=models.CASCADE, related_name='address')
    name = models.CharField(max_length=32, verbose_name='地址名字')
    province = models.ForeignKey('areas.Area', on_delete=models.PROTECT, related_name='province', verbose_name='省')
    city = models.ForeignKey('areas.Area', on_delete=models.PROTECT, related_name='city', verbose_name='市')
    district = models.ForeignKey('areas.Area', on_delete=models.PROTECT, related_name='district', verbose_name='区')
    title = models.CharField(max_length=20, verbose_name='地址名称')
    receiver = models.CharField(max_length=20, verbose_name='收货人')
    place = models.CharField(max_length=50, verbose_name='地址')
    mobile = models.CharField(max_length=11, verbose_name='手机')
    tel = models.CharField(max_length=20, null=True, blank=True, default='', verbose_name='固定电话')
    email = models.CharField(max_length=30, null=True, blank=True, default='', verbose_name='电子邮箱')
    is_deleted = models.BooleanField(default=False, verbose_name='逻辑删除')

    class Meta:
        db_table = 'address'
        verbose_name = '地址'
        verbose_name_plural = verbose_name
        ordering = ['-id']

    def __str__(self):
        return f"{self.name}"
